package com.erp.utils;

import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;

import java.io.*;
import java.net.MalformedURLException;
import java.util.Properties;


public class FtpUtils {
    //ftp服务器地址
    public String hostname = "192.168.1.101";
    //ftp服务器端口号默认为21
    public Integer port = 21 ;
    //ftp登录账号
    public String username = "ftpadmin";
    //ftp登录密码
    public String password = "1234";
    //编码格式
    public String LOCAL_CHARSET="GBK";
    // FTP协议里面，规定文件名编码为iso-8859-1
    public String SERVER_CHARSET = "ISO-8859-1";


    public FTPClient ftpClient = null;

    public FtpUtils() {
        //  this.initFtpClient();
    }

    /**
     * 初始化ftp服务器
     */
    public void initFtpClient() {
        Properties properties = PropertiesUtils.loadProperties("ftp");
        hostname = properties.getProperty("ftpHost");
        port = Integer.parseInt(properties.getProperty("ftpPort"));
        username = properties.getProperty("ftpUserName");
        password = properties.getProperty("ftpPassword");
        LOCAL_CHARSET=properties.getProperty("LOCAL_CHARSET");
        ftpClient = new FTPClient();
        // ftpClient.setControlEncoding(LOCAL_CHARSET);
        try {
            System.out.println("connecting...ftp服务器:"+this.hostname+":"+this.port);
            ftpClient.connect(hostname, port); //连接ftp服务器
            //设置ftp连接模式 被动模式
            ftpClient.enterLocalPassiveMode();

            boolean login = ftpClient.login(username, password);//登录ftp服务器
            int replyCode = ftpClient.getReplyCode(); //是否成功登录服务器
            if(!FTPReply.isPositiveCompletion(replyCode)){
                System.out.println("connect failed...ftp服务器:"+this.hostname+":"+this.port);
            }
            // 开启服务器对UTF-8的支持，如果服务器支持就用UTF-8编码，否则就使用本地编码（GBK）.
            if (FTPReply.isPositiveCompletion(ftpClient.sendCommand("OPTS UTF8", "ON"))) {
                LOCAL_CHARSET = "UTF-8";
            }
            ftpClient.setControlEncoding(LOCAL_CHARSET);
            System.out.println("connect successfu...ftp服务器:"+this.hostname+":"+this.port);
        }catch (MalformedURLException e) {
            e.printStackTrace();
        }catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 上传文件
     * @param pathname ftp服务保存地址
     * @param fileName 上传到ftp的文件名
     *  @param originfilename 待上传文件的名称（绝对地址） *
     * @return
     */
    public boolean uploadFile( String pathname, String fileName,String originfilename){
        boolean flag = true;
        FileOutputStream downloadFile=null;
        FileInputStream inputStream=null;
        String fen = System.getProperty("file.separator");
        try {
            int index;
            byte[] buffer = new byte[1024*10];
            makeDirectory(pathname);
            existFile(pathname+fen+fileName);
            inputStream=new FileInputStream(new File(originfilename));
//            zipUnit.zip(pathname + fen + fileName,inputStream);
            File file1 = new File(pathname + fen + fileName);
            downloadFile = new FileOutputStream(file1);
            int length = 0;
            while ((length = inputStream.read(buffer, 0, buffer.length)) != -1) {
                downloadFile.write(buffer, 0, length);
            }
        }catch (Exception e){
            flag=false;
        }finally {
            try {
                if (null != downloadFile) {
                    downloadFile.flush();
                }
                if (null != downloadFile) {
                    downloadFile.close();
                }
                if (null != inputStream) {
                    inputStream.close();
                }
            }catch (Exception e){

            }
        }
        return flag;
//        boolean flag = false;
//        InputStream inputStream = null;
//        try{
//            System.out.println("开始上传文件");
//            inputStream = new FileInputStream(new File(originfilename));
//            initFtpClient();
//            ftpClient.setFileType(ftpClient.BINARY_FILE_TYPE);
//            CreateDirecroty(pathname);
//            ftpClient.makeDirectory(pathname);
//            ftpClient.changeWorkingDirectory(pathname);
//            ftpClient.enterLocalPassiveMode();
//            boolean b = ftpClient.storeFile(fileName, inputStream);
//            inputStream.close();
//            ftpClient.logout();
//            flag = b;
//            System.out.println("上传文件成功");
//        }catch (Exception e) {
//            System.out.println("上传文件失败");
//            e.printStackTrace();
//        }finally{
//            if(ftpClient.isConnected()){
//                try{
//                    ftpClient.disconnect();
//                }catch(IOException e){
//                    e.printStackTrace();
//                }
//            }
//            if(null != inputStream){
//                try {
//                    inputStream.close();
//                } catch (IOException e) {
//                    e.printStackTrace();
//                }
//            }
//            return flag;
//        }

    }
    /**
     * 上传文件
     * @param pathname ftp服务保存地址
     * @param fileName 上传到ftp的文件名
     *  @param originfilename 待上传文件的名称（绝对地址） *
     * @return
     */
    public boolean ftpUpload(String pathname, String fileName,String originfilename){
        boolean flag = false;
        InputStream inputStream = null;
        try{
            System.out.println("开始上传文件");
            inputStream = new FileInputStream(new File(originfilename));
            initFtpClient();
            ftpClient.setFileType(ftpClient.BINARY_FILE_TYPE);
            CreateDirecroty(pathname);
            ftpClient.makeDirectory(pathname);
            ftpClient.changeWorkingDirectory(pathname);
            ftpClient.enterLocalPassiveMode();
            boolean b = ftpClient.storeFile(fileName, inputStream);
            inputStream.close();
            ftpClient.logout();
            flag = b;
            System.out.println("上传文件成功");
        }catch (Exception e) {
            System.out.println("上传文件失败");
            e.printStackTrace();
        }finally{
            if(ftpClient.isConnected()){
                try{
                    ftpClient.disconnect();
                }catch(IOException e){
                    e.printStackTrace();
                }
            }
            if(null != inputStream){
                try {
                    inputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            return flag;
        }
    }
    /**
     * 上传文件
     * @param pathname ftp服务保存地址
     * @param fileName 上传到ftp的文件名
     * @param inputStream 输入文件流
     * @return
     */
    public boolean uploadFile( String pathname, String fileName,InputStream inputStream){
        boolean flag = true;
        FileOutputStream downloadFile=null;
        String fen = System.getProperty("file.separator");
        try {
            int index;
            byte[] buffer = new byte[1024*10];
            makeDirectory1(pathname);
            existFile1(pathname+fen+fileName);
//            zipUnit.zip(pathname + fen + fileName,inputStream);
            File file1 = new File(pathname + fen + fileName);
            downloadFile = new FileOutputStream(file1);
            int length = 0;
            while ((length = inputStream.read(buffer, 0, buffer.length)) != -1) {
                downloadFile.write(buffer, 0, length);
            }
        }catch (Exception e){
            flag=false;
        }finally {
            try {
                if (null != downloadFile) {
                    downloadFile.flush();
                }
                if (null != downloadFile) {
                    downloadFile.close();
                }
                if (null != inputStream) {
                    inputStream.close();
                }
            }catch (Exception e){

            }
        }
        return flag;
//        boolean flag = false;
//        try{
//            System.out.println("开始上传文件");
//            initFtpClient();
//            ftpClient.setFileType(ftpClient.BINARY_FILE_TYPE);
//            CreateDirecroty(pathname);
//            boolean b1 = ftpClient.makeDirectory(pathname);
//            boolean b2 = ftpClient.changeWorkingDirectory(pathname);
//            ftpClient.enterLocalPassiveMode();
//            boolean b = ftpClient.storeFile(new String(fileName.getBytes(LOCAL_CHARSET),
//                    SERVER_CHARSET), inputStream);
//            inputStream.close();
//            ftpClient.logout();
//            flag = b;
//            System.out.println("上传文件成功");
//        }catch (Exception e) {
//            System.out.println("上传文件失败");
//            e.printStackTrace();
//        }finally{
//            if(ftpClient.isConnected()){
//                try{
//                    ftpClient.disconnect();
//                }catch(IOException e){
//                    e.printStackTrace();
//                }
//            }
//            if(null != inputStream){
//                try {
//                    inputStream.close();
//                } catch (IOException e) {
//                    e.printStackTrace();
//                }
//            }
//            return flag;
//        }
    }
    /**
     * 上传文件
     * @param pathname ftp服务保存地址
     * @param fileName 上传到ftp的文件名
     * @param inputStream 输入文件流
     * @return
     */
    public boolean ftpUpload(String pathname, String fileName,InputStream inputStream){
        boolean flag = false;
        try{
            System.out.println("开始上传文件");
            initFtpClient();
            ftpClient.setFileType(ftpClient.BINARY_FILE_TYPE);
            CreateDirecroty(pathname);
            boolean b1 = ftpClient.makeDirectory(pathname);
            boolean b2 = ftpClient.changeWorkingDirectory(pathname);
            ftpClient.enterLocalPassiveMode();
            boolean b = ftpClient.storeFile(new String(fileName.getBytes(LOCAL_CHARSET),
                    SERVER_CHARSET), inputStream);
            inputStream.close();
            ftpClient.logout();
            flag = b;
            System.out.println("上传文件成功");
        }catch (Exception e) {
            System.out.println("上传文件失败");
            e.printStackTrace();
        }finally{
            if(ftpClient.isConnected()){
                try{
                    ftpClient.disconnect();
                }catch(IOException e){
                    e.printStackTrace();
                }
            }
            if(null != inputStream){
                try {
                    inputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            return flag;
        }

    }
    //创建目录
    public boolean makeDirectory1(String dir) {
        boolean flag = true;
        File file = new File(dir);
        if(file.exists()==false){
            file.mkdirs();
        }
        return flag;
    }
    //判断ftp服务器文件是否存在 ,存在就删除
    public boolean existFile1(String path) throws IOException {
        boolean flag = false;
        File file = new File(path);
        if(file.exists()){
            file.delete();
        }
        return flag;
    }
    //改变目录路径
    public boolean changeWorkingDirectory(String directory) {
        boolean flag = false;
        try {
            flag = ftpClient.changeWorkingDirectory(directory);
            if (flag) {
                System.out.println("进入文件夹" + directory + " 成功！");

            } else {
                System.out.println("进入文件夹" + directory + " 失败！开始创建文件夹");
            }
        } catch (IOException ioe) {
            ioe.printStackTrace();
        }finally {
            return flag;
        }

    }

    //创建多层目录文件，如果有ftp服务器已存在该文件，则不创建，如果无，则创建
    public boolean CreateDirecroty(String remote) throws IOException {
        boolean success = true;
        String directory = remote + "/";
        // 如果远程目录不存在，则递归创建远程服务器目录
        if (!directory.equalsIgnoreCase("/") && !changeWorkingDirectory(new String(directory))) {
            int start = 0;
            int end = 0;
            if (directory.startsWith("/")) {
                start = 1;
            } else {
                start = 0;
            }
            end = directory.indexOf("/", start);
            String path = "";
            String paths = "";
            while (true) {
                String subDirectory = remote.substring(start, end);
                path = path + "/" + subDirectory;
                if (!existFile(path)) {
                    if (makeDirectory(subDirectory)) {
                        changeWorkingDirectory(subDirectory);
                    } else {
                        System.out.println("创建目录[" + subDirectory + "]失败");
                        changeWorkingDirectory(subDirectory);
                    }
                } else {
                    changeWorkingDirectory(subDirectory);
                }

                paths = paths + "/" + subDirectory;
                start = end + 1;
                end = directory.indexOf("/", start);
                // 检查所有目录是否创建完毕
                if (end <= start) {
                    break;
                }
            }
        }
        return success;
    }

    //判断ftp服务器文件是否存在
    public boolean existFile(String path) throws IOException {
        boolean flag = false;
        FTPFile[] ftpFileArr = ftpClient.listFiles(path);
        if (ftpFileArr.length > 0) {
            flag = true;
        }
        return flag;
    }
    //创建目录
    public boolean makeDirectory(String dir) {
        boolean flag = true;
        try {
            flag = ftpClient.makeDirectory(dir);
            if (flag) {
                System.out.println("创建文件夹" + dir + " 成功！");

            } else {
                System.out.println("创建文件夹" + dir + " 失败！");
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return flag;
    }

    /** * 下载文件 *
     * @param pathname FTP服务器文件目录 *
     * @param filename 文件名称 *
     * @param localpath 下载后的文件路径 *
     * @return */
    public  boolean downloadFile(String pathname, String filename, String localpath){
        boolean flag = false;
        FileOutputStream os=null;
        try {
            System.out.println("开始下载文件");
            initFtpClient();
            //切换FTP目录
            boolean b = ftpClient.changeWorkingDirectory(pathname);
            if(b){
                FTPFile[] ftpFiles = ftpClient.listFiles();
                for(FTPFile file : ftpFiles){
                    if(filename.equalsIgnoreCase(file.getName())){
                        long size = file.getSize();
                        byte[] buffer = new byte[20480];

                        //  FileInputStream inputStream=new FileInputStream();
                        File localFile = new File(localpath + "/" + file.getName());
                        os = new FileOutputStream(localFile);
                        ftpClient.retrieveFile(file.getName(), os);
                        os.close();
                        break;
                    }
                }
                ftpClient.logout();
                flag = true;
                System.out.println("下载文件成功");
            }else{
                ftpClient.logout();
                flag = false;
                System.out.println("下载文件失败");
            }
        } catch (Exception e) {
            System.out.println("下载文件失败");
            e.printStackTrace();
        } finally{
            if(ftpClient.isConnected()){
                try{
                    ftpClient.disconnect();
                }catch(IOException e){
                    e.printStackTrace();
                }
            }
            if(null != os){
                try {
                    os.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            return flag;
        }

    }
/*

    public  boolean downloadFile(String pathname, String filename, String localpath){
        boolean flag = false;
        FileOutputStream os=null;
        try {
            System.out.println("开始下载文件");
            initFtpClient();
            //切换FTP目录
            ftpClient.changeWorkingDirectory(pathname);
            FTPFile[] ftpFiles = ftpClient.listFiles();
            for(FTPFile file : ftpFiles){
                if(filename.equalsIgnoreCase(file.getName())){
                    File localFile = new File(localpath + "/" + file.getName());
                    os = new FileOutputStream(localFile);
                    ftpClient.retrieveFile(file.getName(), os);
                    os.close();
                    break;
                }
            }
            ftpClient.logout();
            flag = true;
            System.out.println("下载文件成功");
        } catch (Exception e) {
            System.out.println("下载文件失败");
            e.printStackTrace();
        } finally{
            if(ftpClient.isConnected()){
                try{
                    ftpClient.disconnect();
                }catch(IOException e){
                    e.printStackTrace();
                }
            }
            if(null != os){
                try {
                    os.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            return flag;
        }

    }
*/

    /** * 删除文件 * 
     * @param pathname FTP服务器保存目录 *
     * @param filename 要删除的文件名称 *
     * @return */
    public boolean deleteFile(String pathname, String filename){
        boolean flag = false;
        try {
            System.out.println("开始删除文件");
            initFtpClient();
            //切换FTP目录 
            ftpClient.changeWorkingDirectory(pathname);
            ftpClient.dele(filename);
            ftpClient.logout();
            flag = true;
            System.out.println("删除文件成功");
        } catch (Exception e) {
            System.out.println("删除文件失败");
            e.printStackTrace();
        } finally {
            if(ftpClient.isConnected()){
                try{
                    ftpClient.disconnect();
                }catch(IOException e){
                    e.printStackTrace();
                }
            }
            return flag;
        }

    }

}
